package cn.ld.module.system.dal.mapper.oauth2;

import cn.ld.framework.common.pojo.PageResult;
import cn.ld.framework.mybatis.core.mapper.BaseMapperX;
import cn.ld.framework.mybatis.core.query.LambdaQueryWrapperX;
import cn.ld.framework.tenant.core.aop.TenantIgnore;
import cn.ld.module.system.controller.admin.oauth2.vo.token.OAuth2AccessTokenPageInVO;
import cn.ld.module.system.dal.entity.oauth2.OAuth2AccessTokenEntity;
import org.apache.ibatis.annotations.Mapper;

import java.time.LocalDateTime;
import java.util.List;

@Mapper
public interface OAuth2AccessTokenMapper extends BaseMapperX<OAuth2AccessTokenEntity> {

    @TenantIgnore // 获取 token 的时候，需要忽略租户编号。原因是：一些场景下，可能不会传递 tenant-id 请求头，例如说文件上传、积木报表等等
    default OAuth2AccessTokenEntity selectByAccessToken(String accessToken) {
        return selectOne(OAuth2AccessTokenEntity::getAccessToken, accessToken);
    }

    default List<OAuth2AccessTokenEntity> selectListByRefreshToken(String refreshToken) {
        return selectList(OAuth2AccessTokenEntity::getRefreshToken, refreshToken);
    }

    default PageResult<OAuth2AccessTokenEntity> selectPage(OAuth2AccessTokenPageInVO inVO) {
        return selectPage(inVO, new LambdaQueryWrapperX<OAuth2AccessTokenEntity>()
                .eqIfPresent(OAuth2AccessTokenEntity::getUserId, inVO.getUserId())
                .eqIfPresent(OAuth2AccessTokenEntity::getUserType, inVO.getUserType())
                .likeIfPresent(OAuth2AccessTokenEntity::getClientId, inVO.getClientId())
                .gt(OAuth2AccessTokenEntity::getExpiresTime, LocalDateTime.now())
                .orderByDesc(OAuth2AccessTokenEntity::getId));
    }

}
